package cn.easysan.algorithm.linear;

import java.util.Iterator;

/**
 * @author 13545
 * @Description: 单向链表
 * @date 2021/4/24 19:33
 */
public class LinkList<T> implements Iterable{
    //记录头节点
    private Node head;
    //记录链表的长度
    private int N;

    //节点类
    private class Node{
        //存储数据
        T item;
        //下一个节点
        Node next;

        public Node(){}

        public Node(T item, Node next){
            this.item = item;
            this.next = next;
        }
    }

    public LinkList(){
        //初始化头节点
        this.head = new Node(null,null);
        //初始化元素个数
        this.N = 0;
    }

    //清空链表
    public void clear(){
        head.next = null;
        this.N = 0;
    }

    //获取链表的长度
    public int length(){
        return N;
    }

    //判断链表是否为空
    public boolean isEmpty(){
        return N == 0;
    }

    //获取指定位置i处的元素
    public T get(int i){
        //通过循环，从头节点开始往后找，依次找i次，就可以找到对应的元素
        Node n = head.next;
        for(int index=0;index<i;index++){
            n = n.next;
        }
        return n.item;
    }

    //向链表中添加元素t
    public void insert(T t){
        //找到最后一个节点
        Node n = head;
        while (n.next != null){
            n = n.next;
        }
        //穿件新节点，保存元素t
        Node newNode = new Node(t, null);
        //让当前最后一个节点指向新节点
        n.next = newNode;
        //元素个数+1
        N++;
    }

    //向指定位置i处，添加元素
    public void insert(int i, T t){
        //找到i位置前一个节点
        Node pre = head;
        for(int index=0;index <= i-1;index++){
            pre = pre.next;
        }
        //找到i位置的节点
        Node curr = pre.next;
        //创建新节点，并且新节点需要指向原来i位置的节点
        Node newNode = new Node(t, curr);
        //原来i位置的前一个节点指向新节点即可
        pre.next = newNode;
        //元素的个数+1
        N++;
    }

    //删除指定位置i处的元素，并返回被删除的元素
    public T remove(int i){
        //找到i位置的前一个节点
        Node pre = this.head;
        for(int index = 0;index <= i-1;index++){
            pre = pre.next;
        }
        //找到i位置的节点
        Node curr = pre.next;
        //找到i位置的下一个节点
        Node nextNode = curr.next;
        //前一个节点指向下一个节点
        pre.next = nextNode;
        //元素个数-1
        N--;
        return curr.item;
    }

    //查找元素t在链表中第一次出现的位置
    public int indexOf(T t){
        //从头节点开始，依次找到每一个节点，取出item和t比较，如果相同，就找到了
        Node n = head;
        for(int i=0;n.next != null;i++){
            n = n.next;
            if(n.item.equals(t)){
                return i;
            }
        }
        return -1;
    }

    @Override
    public Iterator iterator() {
        return new LIterator();
    }

    private class LIterator implements Iterator{

        private Node n;

        private LIterator(){
            this.n = head;
        }

        @Override
        public boolean hasNext() {
            return n.next != null;
        }

        @Override
        public Object next() {
            n = n.next;
            return n.item;
        }
    }
}
